import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
file_path = os.path.dirname(__file__)
import tensorflow as tf
import numpy as np
x = np.random.uniform(-1, 1, [2, 3, 1, 54, 3,32]).astype(np.float16)
# golden: np.ndarray = tf.reduce_sum(x, [0, 2, 3])
golden: np.ndarray = tf.reduce_sum(x)
# print(golden.numpy())